package cn.iocoder.yudao.module.vr.service.testgap;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.module.vr.controller.admin.test.vo.TestGap;
import cn.iocoder.yudao.module.vr.controller.admin.testgap.vo.*;
import cn.iocoder.yudao.module.vr.convert.testgap.TestGapConvert;
import cn.iocoder.yudao.module.vr.dal.dataobject.testgap.TestGapDO;
import cn.iocoder.yudao.module.vr.dal.mysql.testgap.TestGapMapper;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

import javax.annotation.Resource;
import java.util.*;
import java.util.stream.Collectors;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.vr.enums.ErrorCodeConstants.*;

/**
 * 试题填空 Service 实现类
 *
 * @author 芋道源码
 */
@Service
@Validated
public class TestGapServiceImpl implements TestGapService {

    @Resource
    private TestGapMapper testGapMapper;

    @Override
    public Long createTestGap(TestGapCreateReqVO createReqVO) {
        // 插入
        TestGapDO testGap = TestGapConvert.INSTANCE.convert(createReqVO);
        testGapMapper.insert(testGap);
        // 返回
        return testGap.getId();
    }

    @Override
    public void createOptionsFromTest(Long testId, List<TestGap> testGaps) {
        for (TestGap testGap : testGaps) {
            TestGapDO gapDO = new TestGapDO();
            gapDO.setTestId(testId);
            gapDO.setOrderId(testGap.getOrderId());
            gapDO.setContent(testGap.getContent());
            gapDO.setScore(testGap.getScore());
            gapDO.setRule(testGap.getRule());
            testGapMapper.insert(gapDO);
        }
    }

    @Override
    public void updateTestGap(TestGapUpdateReqVO updateReqVO) {
        // 校验存在
        validateTestGapExists(updateReqVO.getId());
        // 更新
        TestGapDO updateObj = TestGapConvert.INSTANCE.convert(updateReqVO);
        testGapMapper.updateById(updateObj);
    }

    @Override
    public void updateTestGapsFromTest(Long testId, List<TestGap> testGaps) {
        // 获取该试题的所有填空
        List<TestGapDO> gapDOList = testGapMapper.selectListByTestId(testId);
        Map<Long, TestGapDO> gapDOMap = gapDOList.stream().collect(Collectors.toMap(TestGapDO::getId, gap -> gap));
        List<TestGap> createGaps = new ArrayList<>();
        for (TestGap testGap : testGaps) {
            if (testGap.getId() == null || testGap.getId() == 0) {
                // 新增
                createGaps.add(testGap);
            } else {
                validateTestGapExists(testGap.getId());
                TestGapDO gapDO = gapDOMap.get(testGap.getId());
                gapDO.setContent(testGap.getContent());
                gapDO.setScore(testGap.getScore());
                gapDO.setOrderId(testGap.getOrderId());
                gapDO.setRule(testGap.getRule());
                testGapMapper.updateById(gapDO);
                // 移除已操作的gap
                gapDOMap.remove(testGap.getId());
            }
        }
        if (!createGaps.isEmpty()) {
            createOptionsFromTest(testId, createGaps);
        }
        if (!gapDOMap.isEmpty()) {
            gapDOMap.keySet().forEach(this::deleteTestGap);
        }
    }

    @Override
    public List<TestGap> getTestGapListByTestId(Long testId) {
        List<TestGapDO> testGapDOS = testGapMapper.selectListByTestId(testId);
        List<TestGap> testGaps = new ArrayList<>();
        for (TestGapDO testGapDO : testGapDOS) {
            TestGap testGap = new TestGap(testGapDO.getId()
                    , testGapDO.getTestId()
                    , testGapDO.getOrderId()
                    , testGapDO.getContent()
                    , testGapDO.getScore()
                    , testGapDO.getRule());
            testGaps.add(testGap);
        }
        testGaps.sort(Comparator.comparing(TestGap::getOrderId));
        return testGaps;
    }

    @Override
    public void deleteTestGap(Long id) {
        // 校验存在
        validateTestGapExists(id);
        // 删除
        testGapMapper.deleteById(id);
    }

    private void validateTestGapExists(Long id) {
        if (testGapMapper.selectById(id) == null) {
            throw exception(TEST_NOT_EXISTS);
        }
    }

    @Override
    public TestGapDO getTestGap(Long id) {
        return testGapMapper.selectById(id);
    }

    @Override
    public List<TestGapDO> getTestGapList(Collection<Long> ids) {
        return testGapMapper.selectBatchIds(ids);
    }

    @Override
    public PageResult<TestGapDO> getTestGapPage(TestGapPageReqVO pageReqVO) {
        return testGapMapper.selectPage(pageReqVO);
    }

    @Override
    public List<TestGapDO> getTestGapList(TestGapExportReqVO exportReqVO) {
        return testGapMapper.selectList(exportReqVO);
    }

}
